#include "login.h"
#include <stdio.h>
#include <string.h>

int menu(FILE* fp)
{
	puts("***************");
	puts("1.  用户登陆");
	puts("2.  用户注册");
	puts("3.  退出程序");
	puts("***************");
	printf("\n    请选择:");
	int c = 0;
	scanf("%d",&c);
	
	switch(c)
	{
		case 1:
			login(fp);
			break;
		case 2:
			reg(fp);
			break;
		case 3:
			break;
		default:	
			puts("选择有误，请重新选择!");
			break;
	}
	return c;
}

int login(FILE* fp)
{
	DATA data,data2;
	while(getchar() != '\n');
	printf("请输入账号:");
	scanf("%s",data.number);
	while(getchar() != '\n');
	printf("请输入密码:");
	scanf("%s",data.password);
	
	rewind(fp);	
	while(fscanf(fp, "%s %s ", data2.number, data2.password) != -1)
	{
		if((strcmp(data.number,data2.number) == 0) && (strcmp(data.password,data2.password) == 0))
		{
			puts("登陆成功!");
			return 0;
		}
	}
	
	puts("用户名与密码不匹配，请重试!");
	return -1;
	
}

void reg(FILE* fp)
{
	DATA data;
	while(getchar() != '\n');
	printf("请输入账号:");
	scanf("%s",data.number);
	while(getchar() != '\n');
	printf("请输入密码:");
	scanf("%s",data.password);
	
	fprintf(fp, "%s %s ", data.number,data.password);
	puts("注册成功!");
	return ;
}
